# # -*- coding: utf-8 -*-
# import os
# import pandas as pd

# # 获取当前文件夹的路径
# current_dir = os.path.dirname(os.path.abspath(__file__))

# # 查找当前文件夹中的所有CSV文件
# csv_files = [f for f in os.listdir(current_dir) if f.endswith('.csv')]

# # 创建一个空的DataFrame，用于存放数据
# combined_data = pd.DataFrame()

# # 遍历CSV文件列表，读取数据并追加到combined_data中
# for csv_file in csv_files:
#     print(os.path.join(current_dir, csv_file))
#     data = pd.read_csv(os.path.join(current_dir, csv_file))
#     combined_data = combined_data.append(data, ignore_index=True)

# # 保存合并后的数据为XLSX文件
# combined_data.to_excel(os.path.join(current_dir, '合并表格.xlsx'), index=False)

from os import getcwd,listdir,path
import pandas as pd
from time import sleep

# 获取当前目录
current_dir = getcwd()

# 获取当前目录下的所有CSV文件
csv_files = [f for f in listdir(current_dir) if f.endswith(".csv")]

# 创建一个空的DataFrame用于存储合并后的数据
merged_data = pd.DataFrame()

# 接受用户输入选择编码方式
print("请选择编码方式：")
print("1. GBK")
print("2. UTF-8")
print("3. ISO-8859-1")
encoding_choice = input()

# 读取每个CSV文件并将数据合并到DataFrame中
for csv_file in csv_files:
    file_path = path.join(current_dir, csv_file)
    # file_path = current_dir+'\\'+ csv_file
    print(">>>>>>>> 合并: " + file_path)
    if encoding_choice == "1":
        data = pd.read_csv(file_path, encoding="GBK")
    elif encoding_choice == "2":
        data = pd.read_csv(file_path, encoding="utf-8")
    elif encoding_choice == "3":
        data = pd.read_csv(file_path, encoding="ISO-8859-1")
    else:
        print("无效的选择！")
        break
    merged_data = pd.concat([merged_data, data], ignore_index=True)

# 将合并后的数据写入到XLSX文件
output_file = os.path.join(current_dir, "merged_data.xlsx")
merged_data.to_excel(output_file, index=False)

print("合并完成，保存为merged_data.xlsx文件。")

sleep(1000)